import dotenv
from langchain_community.chat_models import ChatOpenAI
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

dotenv.load_dotenv()

    # 构建提示词
prompt = (ChatPromptTemplate
          .from_template("你是OpenAi研发的机器人，请按照人类的要求来回答问题，人类的问题是：{query}"))
    # 创建modle

llm = ChatOpenAI(model="gpt-3.5-turbo-16k")
    # 创建parser
parser = StrOutputParser()
    # 构建链接
chain = prompt|llm|parser
    # 调用链
print(chain.invoke({"query","请讲一个医生的笑话"}))
